#include "BlockQueue.hpp"
#include <ctime>
#include <sys/types.h>
#include <unistd.h>
#include "Task.hpp"



void *productor(void *p_bqs_) // 生产
{
    BlockQueue<CalTask> *p_c_bq = static_cast<BlockQueues<CalTask,SaveTask> *>(p_bqs_)->_p_c_bq;//类型转换的优先级高

    while (true)
    {
        // 生产任务
        int x = rand() % 10 + 1; // 随机1~10的数
        int y = rand() % 10 + 1; // 随机1~10的数
        int operCode = rand()%oper.size();
        CalTask t(x, y, oper[operCode] , mymath);
        p_c_bq->push(t);
        std::cout<<"productor-生产任务 ： " << t.toString() << std::endl;
    }
    return nullptr;
}

void *consumer(void *p_bqs_) // 带_是为了区分 消费
{
    BlockQueue<CalTask> *p_c_bq = static_cast<BlockQueues<CalTask,SaveTask> *>(p_bqs_)->_p_c_bq;
    BlockQueue<SaveTask> *p_s_bq = static_cast<BlockQueues<CalTask,SaveTask> *>(p_bqs_)->_p_s_bq;

    while (true)
    {
        // 计算任务
        CalTask t;
        p_c_bq->pop(&t);
        std::cout << "consumer-计算任务 ：" << t() << "...done" << std::endl;
        std::string result = t();

        //将任务放入到用于存储的阻塞队列当中
        SaveTask s(result, Save);
        p_s_bq->push(s);
        std::cout<<"consumer-推送任务 ： " << t() << "完成" << std::endl;
        
        // sleep(1);
    }
    return nullptr;
}

void* saver(void* p_bqs_)
{
    BlockQueue<SaveTask> *p_s_bq = static_cast<BlockQueues<CalTask,SaveTask> *>(p_bqs_)->_p_s_bq;

    while(true)
    {
        SaveTask s;
        p_s_bq->pop(&s);
        s();
        std::cout<<"saver-存储任务完成" << std::endl;
    }

    return nullptr;
}


int main()
{
    srand((unsigned long)time(nullptr) ^ getpid());//用来生成随机数
    BlockQueues<CalTask,SaveTask> bqs;
    bqs._p_c_bq = new BlockQueue<CalTask>(); // 相当于生产消费模型当中的缓冲区（超市）
    bqs._p_s_bq = new BlockQueue<SaveTask>(); // 用于存储任务信息

    //三个线程分别负责生产，计算，存储
    //生产：将任务放入到用于计算的阻塞队列当中（CalBlockQueue）
    //计算：将阻塞队列当中（CalBlockQueue）的任务计算并删除，再将该任务放入到用于存储的阻塞队列当中（SaveBlockQueue）
    //存储：将阻塞队列当中（SaveBlockQueue）的任务存储到特定的文件并删除该任务
    pthread_t c[2], p[3] ,s;                              // c:Consumption 消费  p:Production 生产 s：save 保存
    pthread_create(p, nullptr, productor, &bqs);
    pthread_create(p+1, nullptr, productor, &bqs);
    pthread_create(p+2, nullptr, productor, &bqs);
    pthread_create(c, nullptr, consumer, &bqs); // 传入bqs，让c，p, s线程能够看到"同一份"资源（缓冲区）
    pthread_create(c+1, nullptr, consumer, &bqs); // 传入bqs，让c，p, s线程能够看到"同一份"资源（缓冲区）
    pthread_create(&s, nullptr, saver, &bqs);

    pthread_join(p[0], nullptr);    
    pthread_join(p[1], nullptr);    
    pthread_join(p[2], nullptr);    
    pthread_join(c[0], nullptr);
    pthread_join(c[1], nullptr);
    pthread_join(s, nullptr);


    delete bqs._p_c_bq;
    delete bqs._p_s_bq;

    return 0;
}